AttributeError: 'Model' object has no attribute 'predict

您所在的位置:网站首页 have classes的意思 AttributeError: 'Model' object has no attribute 'predict

AttributeError: 'Model' object has no attribute 'predict

2023-09-11 00:06| 来源: 网络整理| 查看: 265

使用keras训练好了mnist数字识别模型后,准备拿来做预测。 以下是预测代码:

#coding:utf-8 import cv2 import numpy as np from keras.models import load_model model = load_model('self.h5') image = cv2.imread('Predict_image/7.jpg', 0) img = cv2.imread('Predict_image/7.jpg', 0) img = (img.reshape(1,28,28,1)).astype("float32")/255 predict = model.predict_classes(img) print ('识别为:') print (predict) cv2.imshow("Image1", image) cv2.waitKey(0)

之前这个代码是有效的,但是在更换训练模型后,代码出现提示:

AttributeError: 'Model' object has no attribute 'predict_classes'

在这里插入图片描述 在国外网站上找到了解决办法。

The predict_classes method is only available for the Sequential class (which is the class of your first model) but not for the Model class (the class of your second model).

With the Model class, you can use the predict method which will give you a vector of probabilities and then get the argmax of this vector (with np.argmax(y_pred1,axis=1)).

原因很简单,两次训练时,网络结构不同。 第一次用的网络是在model=Sequential()下添加模块的的方法,也就是所谓的顺序模型。Sequential class可以使用model.predict_classes()的方法来实现预测;第二次用的方法,是编写好网络结构后使用model=Model()综合起来的方法,也就是Model class,无法使用model.predict_classes(),但是可以使用组合形式预测。 将

predict = model.predict_classes(img)

改为

predict = model.predict(img) predict=np.argmax(predict,axis=1)

即可。

参考网址: https://stackoverflow.com/questions/44806125/attributeerror-model-object-has-no-attribute-predict-classes 希望本博客对您的问题有所帮助!



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3